from abc import ABC, abstractmethod
from torch.utils.data import Dataset

class BaseDatasetPlugin(ABC):
    """数据集插件基类"""
    
    @abstractmethod
    def create_dataset(self, config):
        """创建数据集实例"""
        pass
        
    @abstractmethod
    def create_dataloader(self, dataset, config):
        """创建数据加载器"""
        pass